
import os
from langchain_tavily import TavilySearch
from db.db_work import get_db
from langchain_core.tools import tool

"""openai参数配置"""
os.environ["TAVILY_API_KEY"] = "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"
"""工具配置"""
# 创建执行数据库sql语句的函数
@tool
def run_sql(query:str):
    """
            执行SQL语句，根据语句类型返回不同结果
            - 数据库中blask_list为黑名单表，其中exit=0表示他以前在黑名单中的，exit=1表示他现在就在黑名单中
            - orders表为未处理的订单表，history_buy中的表为已经处理好了的订单,history_buy表中的comment为对这次订单情况的评分
            -
            - 查询语句(SELECT)：返回查询结果列表，每个元素是一行数据的字典
            - 其他语句：返回布尔值表示执行是否成功
    """
    db = get_db()
    # 执行查询
    result = db.execute(query)
    print(result)
    return result


# 初始化工具（这里添加了Tavily搜索工具）
tools = [TavilySearch(max_results=2),run_sql]
# "SELECT * FROM products WHERE price < 100"

# run_sql("SELECT * FROM products WHERE price < 100")